../../../../../models/demos/wormhole/stable_diffusion/tests/vae/test_vae.py